Pro tvorbu TestNG testovací třídy je nutné doplnit balíčky a upravit některá pole ve skriptu. Žádné větší změny zde nejsou. Akorát se zde přidá navíc BeforeMethod a AfterMethod, která je trochu upravená. Testovací metoda nyní umožňuje vkládání priority a aktivity.

import org.testng.annotations.Test;

import org.testng.annotations.BeforeMethod;

import org.testng.ITestResult;

import org.testng.annotations.AfterMethod;

public ITestResult result;

@BeforeMethod

public void beforeMethod() throws Exception{

}

@Test // Může tu být nyní priorita a aktivita

@AfterMethod

public void afterMethod(ITestResult result) {

                  driver.quit();

                  String verificationErrorString = verificationErrors.toString();

                    if (!"".equals(verificationErrorString)) {

                      fail(verificationErrorString);

                    }

                    try

                    {

                       if(result.getStatus() == ITestResult.SUCCESS)

                       {

                           System.out.println("Test ÚSPĚŠNÝ **********");

                       }

                       else if(result.getStatus() == ITestResult.FAILURE)

                       {

                                   System.out.println("Test NEÚSPĚŠNÝ ***********");

                       }

                        else if(result.getStatus() == ITestResult.SKIP ){

                           System.out.println("Test PŘESKOČEN ***********");

                       }

                    }

                    catch(Exception e)

                    {

                      e.printStackTrace();

                    }

}